package com.java.comprehensive.arithmetic.leecode.easy;

import util.ListNode;

/**
 * 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
 */
public class LeetCode_203 {

    /**
     * 使用假节点的方式循环移除节点
     *
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElements(ListNode head, int val) {
        if (head == null) return null;
        //虚拟头节点
        ListNode dummy = new ListNode();
        dummy.next = head;
        // 用来删除的临时节点
        ListNode curr = dummy;
        while (curr.next != null) {
            if (curr.next.val == val) {
                curr.next = curr.next.next;
            } else {
                curr = curr.next;
            }
        }
        return dummy.next;
    }

    /**
     * 不使用假节点的方式循环移除节点
     *
     * @param head
     * @param val
     * @return
     */
    public ListNode removeElementsTwo(ListNode head, int val) {
        if (head == null) return null;
        while (head != null && head.val == val) {
            head = head.next;
        }
        ListNode curr = head;
        while (curr != null && curr.next != null) {
            if (curr.next.val == val) {
                curr.next = curr.next.next;
            } else {
                curr = curr.next;
            }
        }
        return head;
    }
}
